package com.bolang.onlycode.demo;

import java.util.Arrays;

public class NationalFlag {
  //  例如，给定数组：[2, 3, 1, 9, 7, 6, 1, 4, 5]，给定一个值4，
    //  那么经过处理原数组可能得一种情况是：[2, 3, 1, 1, 4, 9, 7, 6, 5]，
    //  需要注意的是，小于4的部分不需要有序，大于4的部分也不需要有序，返回等于4部分的左右两个下标，即[4, 4]
       public static int[] partitionThree(int[] arr,int l,int r,int p){
           int less = l - 1;
           int more = r + 1;
           while (l<more){
               if(arr[l]<p){
                   swap(arr,++less,l++);
               } else if (arr[l] > p) {
                   swap(arr, --more, l);
               }
               else {
                   l++;
               }
           }
           return new int[] { less+1 , more-1  };
       }
    public static void swap(int[] arr, int i, int j) {
        int temp = arr[i];
        arr[i] = arr[j];
        arr[j] = temp;
    }

    public static void main(String[] args) {
      int[] arr={2,3,1,9,7,6,1,4,5};
        int[] ints = partitionThree(arr, 0, arr.length-1, 4);
        System.out.println(Arrays.toString(ints));


    }
}
